package com.cw.mp.service.impl;

import cn.hutool.core.bean.BeanUtil;
import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.util.StrUtil;
import com.baomidou.mybatisplus.core.metadata.OrderItem;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.baomidou.mybatisplus.extension.toolkit.Db;
import com.cw.mp.domain.dto.PageDTO;
import com.cw.mp.domain.po.Address;
import com.cw.mp.domain.po.User;
import com.cw.mp.domain.vo.AddressVO;
import com.cw.mp.domain.vo.UserVO;
import com.cw.mp.enums.UserStatus;
import com.cw.mp.mapper.UserMapper;
import com.cw.mp.query.UserQuery;
import com.cw.mp.service.UserService;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;

import java.util.*;
import java.util.stream.Collectors;


@Service
public class UserServiceImpl extends ServiceImpl<UserMapper, User> implements UserService {
    // 注意：此处已经注入了UserService无需注入，而且ServiceImpl内部也注入了BaseMapper，所以这里压根不需要注入
    @Override
    @Transactional
    public void deductionBalance(Long id, Long money) {
        // 1. 查询用户
        User user = this.getById(id);
        // 2. 校验用户状态
        if (user == null || user.getStatus() == UserStatus.FROZEN) {
            throw new RuntimeException("用户状态异常");
        }
        // 3. 校验余额是否充足
        if (user.getBalance() < money) {
            throw new RuntimeException("用户余额不足");
        }
        // 4. 扣减余额
//        baseMapper.deductionBalance(id, money);
        // 同样使用lambda方式
        Long remainBalance = user.getBalance() - money;
        lambdaUpdate()
                .set(User::getBalance, remainBalance)
                .set(remainBalance == 0, User::getStatus, UserStatus.FROZEN)
                .eq(User::getId, id)
                .eq(User::getBalance, user.getBalance()) // 乐观锁
                .update(); // 这里一定要写，不写是不会执行的
    }

    //    复杂查询
    @Override
    public List<User> queryUsers(String name, Integer status, Integer minBalance, Integer maxBalance) {
        // 使用lambda查询可直接构建条件无需创建Wrapper对象
        return lambdaQuery() // 属于IService中的方法,通过反射获取字节码文件/表信息
                .like(name != null, User::getUsername, name)
                .eq(status != null, User::getStatus, status)
                .ge(minBalance != null, User::getBalance, minBalance)
                .le(maxBalance != null, User::getBalance, maxBalance)
                .list();
    }

    /**
     * @author: 江传文
     * @date: 2024/6/9 16:24
     * Db 工具类
     * 在你最初的代码中使用 Db.lambdaQuery 可能是因为你需要进行一些不在 ServiceImpl 或 UserService 中的查询操作，
     * 或者是查询的对象不是当前 ServiceImpl 的泛型类型。这种情况下，
     * 可以使用 MyBatis-Plus 提供的 Db 工具类，它可以在任意地方使用静态方法进行查询。
     */
    @Override
    public UserVO queryUserAndAddressById(Long id) {
        // 1. 查询用户
        User user = getById(id);
        if (user == null || user.getStatus() == UserStatus.FROZEN) { // 校验用户
            throw new RuntimeException("用户状态异常");
        }
        // 2. 查询地址->PO
        List<Address> addresses = Db.lambdaQuery(Address.class)
                .eq(Address::getUserId, id)
                .list();
        // 3. 封装VO
        // 3.1 转User的PO为VO
        UserVO userVO = BeanUtil.copyProperties(user, UserVO.class);
        // 3.2 转地址VO
        if (CollUtil.isNotEmpty(addresses)) {
            userVO.setAddresses(BeanUtil.copyToList(addresses, AddressVO.class));
        }
        return userVO;
    }

    @Override
    public List<UserVO> queryUserAndAddressByIds(List<Long> ids) {
        // 1. 查询用户
        List<User> users = listByIds(ids);
        if (CollUtil.isEmpty(users)) {
            return Collections.emptyList();
        }

        // 2. 查询地址
        // 2.1 获取用户id集合
        List<Long> collect = users.stream().map(User::getId).toList();
        // 2.2 根据用户id查询地址
        List<Address> addresses = Db.lambdaQuery(Address.class)
                .in(Address::getUserId, collect)
                .list();
        // 2.3 转换地址VO
        List<AddressVO> addressVOS = BeanUtil.copyToList(addresses, AddressVO.class);
        // 2.4 用户地址及和分组处理，相同用户的放入一个集合(组)中,
        // 使用Stream流的groupingBy方法将同一user_id的地址放至map集合中
        Map<Long, List<AddressVO>> addressMap = new HashMap<>(0);
        if (CollUtil.isNotEmpty(addressVOS)) {
            addressMap = addressVOS.stream().collect(Collectors.groupingBy(AddressVO::getUserId));

        }

        // 3. 转换VO返回
        List<UserVO> list = new ArrayList<>(users.size());
        for (User user : users) {
            // 3.1 转换User的PO为VO
            UserVO userVO = BeanUtil.copyProperties(user, UserVO.class);
            list.add(userVO);
            // 3.2 转换地址VO
            userVO.setAddresses(addressMap.get(user.getId()));
        }
        return list;
    }

    @Override
    public PageDTO<UserVO> queryUsersPage(UserQuery userQuery) {
        String name = userQuery.getName();
        Integer status = userQuery.getStatus();
        // 1. 构建分页条件
        /*// 分页条件
        Page<User> page = Page.of(userQuery.getPageNo(), userQuery.getPageSize());
        // 排序条件
        if (StrUtil.isNotEmpty(userQuery.getSortBy())) {
            page.addOrder(new OrderItem(userQuery.getSortBy(), userQuery.getIsAsc()));
        }
        // 默认按照更新时间排序
        page.addOrder(new OrderItem("update_time", false));*/

        Page<User> page = userQuery.toMpPageDefaultSortByUpdateTimeDesc();
        // 分页查询
        Page<User> p = lambdaQuery()
                .like(name != null, User::getUsername, name)
                .eq(status != null, User::getStatus, status)
                .page(page);


        /*// 3. 封装VO结果
        PageDTO<UserVO> dto = new PageDTO<>();
        // 总条数
        dto.setTotal(p.getTotal());
        // 总页数
        dto.setPages(p.getPages());
        // 当前页数据
        List<User> records = p.getRecords();
        if (CollUtil.isEmpty(records)) {
            dto.setList(Collections.emptyList());
        }
        // 拷贝user的VO
        List<UserVO> list = BeanUtil.copyToList(records, UserVO.class);
        dto.setList(list);
        return dto;*/
//         return PageDTO.of(p,user -> BeanUtil.copyProperties(user, UserVO.class));
         return PageDTO.of(p,user -> {
             // 1. 拷贝基础属性
             UserVO vo = BeanUtil.copyProperties(user, UserVO.class);
             // 2. 处理特殊逻辑
             // 将用户名后两位隐藏
             vo.setUsername(vo.getUsername().substring(0,vo.getUsername().length() - 2)+ "**");
             return vo;
         });
    }
}
